home *** CD-ROM | disk | FTP | other *** search
/ PC World Komputer 2001 December / pcwk12201b.iso / Wersje pelne i specjalne / Winamp 2.77 i 3.0beta / wasabi-sdk_beta1.exe / jnetlib / connection.cpp < prev    next >
C/C++ Source or Header  |  2001-10-08  |  12KB  |  476 lines
  1. /*
  2.  
  3.   Nullsoft WASABI Source File License
  4.  
  5.   Copyright 1999-2001 Nullsoft, Inc.
  6.  
  7.     This software is provided 'as-is', without any express or implied
  8.     warranty.  In no event will the authors be held liable for any damages
  9.     arising from the use of this software.
  10.  
  11.     Permission is granted to anyone to use this software for any purpose,
  12.     including commercial applications, and to alter it and redistribute it
  13.     freely, subject to the following restrictions:
  14.  
  15.     1. The origin of this software must not be misrepresented; you must not
  16.        claim that you wrote the original software. If you use this software
  17.        in a product, an acknowledgment in the product documentation would be
  18.        appreciated but is not required.
  19.     2. Altered source versions must be plainly marked as such, and must not be
  20.        misrepresented as being the original software.
  21.     3. This notice may not be removed or altered from any source distribution.
  22.  
  23.  
  24.   Brennan Underwood
  25.   brennan@nullsoft.com
  26.  
  27. */
  28.  
  29. /*
  30. ** JNetLib
  31. ** Copyright (C) 2000-2001 Nullsoft, Inc.
  32. ** Author: Justin Frankel
  33. ** File: connection.cpp - JNL TCP connection implementation
  34. ** License: see jnetlib.h
  35. */
  36.  
  37. #include "netinc.h"
  38. #include "util.h"
  39. #include "connection.h"
  40.  
  41.  
  42. JNL_Connection::JNL_Connection(JNL_AsyncDNS *dns, int sendbufsize, int recvbufsize)
  43. {
  44.   m_errorstr="";
  45.   if (dns == JNL_CONNECTION_AUTODNS)
  46.   {
  47.     m_dns=new JNL_AsyncDNS();
  48.     m_dns_owned=1;
  49.   }
  50.   else
  51.   {
  52.     m_dns=dns;
  53.     m_dns_owned=0;
  54.   }
  55.   m_recv_buffer_len=recvbufsize;
  56.   m_send_buffer_len=sendbufsize;
  57.   m_recv_buffer=(char*)malloc(m_recv_buffer_len);
  58.   m_send_buffer=(char*)malloc(m_send_buffer_len);
  59.   m_socket=-1;
  60.   memset(m_recv_buffer,0,recvbufsize);
  61.   memset(m_send_buffer,0,sendbufsize);
  62.   m_remote_port=0;
  63.   m_state=STATE_NOCONNECTION;
  64.   m_recv_len=m_recv_pos=0;
  65.   m_send_len=m_send_pos=0;
  66.   m_host[0]=0;
  67.   memset(&m_saddr,0,sizeof(m_saddr));
  68. }
  69.  
  70. void JNL_Connection::connect(int s, struct sockaddr_in *loc)
  71. {
  72.   close(1);
  73.   m_socket=s;
  74.   m_remote_port=0;
  75.   m_dns=NULL;
  76.   if (loc) m_saddr=*loc;
  77.   else memset(&m_saddr,0,sizeof(m_saddr));
  78.   if (m_socket != -1)
  79.   {
  80.     SET_SOCK_BLOCK(m_socket,0);
  81.     m_state=STATE_CONNECTED;
  82.   }
  83.   else 
  84.   {
  85.     m_errorstr="invalid socket passed to connect";
  86.     m_state=STATE_ERROR;
  87.   }
  88. }
  89.  
  90. void JNL_Connection::connect(char *hostname, int port)
  91. {
  92.   close(1);
  93.   m_remote_port=(short)port;
  94.   m_socket=::socket(AF_INET,SOCK_STREAM,0);
  95.   if (m_socket==-1)
  96.   {
  97.     m_errorstr="creating socket";
  98.     m_state=STATE_ERROR;
  99.   }
  100.   else
  101.   {
  102.     SET_SOCK_BLOCK(m_socket,0);
  103.     strncpy(m_host,hostname,sizeof(m_host)-1);
  104.     m_host[sizeof(m_host)-1]=0;
  105.     memset(&m_saddr,0,sizeof(m_saddr));
  106.     if (!m_host[0])
  107.     {
  108.       m_errorstr="empty hostname";
  109.       m_state=STATE_ERROR;
  110.     }
  111.     else
  112.     {
  113.       m_state=STATE_RESOLVING;
  114.           m_saddr.sin_family=AF_INET;
  115.       m_saddr.sin_port=htons((unsigned short)port);
  116.       m_saddr.sin_addr.s_addr=inet_addr(hostname);
  117.     }
  118.   }
  119. }
  120.  
  121. JNL_Connection::~JNL_Connection()
  122. {
  123.   if (m_socket >= 0)
  124.   {
  125.     ::shutdown(m_socket, SHUT_RDWR);
  126.     ::closesocket(m_socket);
  127.     m_socket=-1;
  128.   }
  129.   free(m_recv_buffer);
  130.   free(m_send_buffer);
  131.   if (m_dns_owned) 
  132.   {
  133.     delete m_dns;
  134.   }
  135. }
  136.  
  137. void JNL_Connection::run(int max_send_bytes, int max_recv_bytes, int *bytes_sent, int *bytes_rcvd)
  138. {
  139.   int bytes_allowed_to_send=(max_send_bytes<0)?m_send_buffer_len:max_send_bytes;
  140.   int bytes_allowed_to_recv=(max_recv_bytes<0)?m_recv_buffer_len:max_recv_bytes;
  141.  
  142.   if (bytes_sent) *bytes_sent=0;
  143.   if (bytes_rcvd) *bytes_rcvd=0;
  144.  
  145.   switch (m_state)
  146.   {
  147.     case STATE_RESOLVING:
  148.       if (m_saddr.sin_addr.s_addr == INADDR_NONE)
  149.       {
  150.         int a=m_dns?m_dns->resolve(m_host,(unsigned long int *)&m_saddr.sin_addr.s_addr):-1;
  151.         if (!a) { m_state=STATE_CONNECTING; }
  152.         else if (a == 1)
  153.         {
  154.           m_state=STATE_RESOLVING; 
  155.           break;
  156.         }
  157.         else
  158.         {
  159.           m_errorstr="resolving hostname"; 
  160.           m_state=STATE_ERROR; 
  161.           return;
  162.         }
  163.       }
  164.       if (!::connect(m_socket,(struct sockaddr *)&m_saddr,16)) 
  165.       {
  166.         m_state=STATE_CONNECTED;
  167.       }
  168.       else if (ERRNO!=EINPROGRESS)
  169.       {
  170.         m_errorstr="connecting to host";
  171.         m_state=STATE_ERROR;
  172.       }
  173.       else { m_state=STATE_CONNECTING; }
  174.     break;
  175.     case STATE_CONNECTING:
  176.       {        
  177.         fd_set f[3];
  178.         FD_ZERO(&f[0]);
  179.         FD_ZERO(&f[1]);
  180.         FD_ZERO(&f[2]);
  181.         FD_SET(m_socket,&f[0]);
  182.         FD_SET(m_socket,&f[1]);
  183.         FD_SET(m_socket,&f[2]);
  184.         struct timeval tv;
  185.         memset(&tv,0,sizeof(tv));
  186.         if (select(m_socket+1,&f[0],&f[1],&f[2],&tv)==-1)
  187.         {
  188.           m_errorstr="connecting to host (calling select())";
  189.           m_state=STATE_ERROR;
  190.         }
  191.         else if (FD_ISSET(m_socket,&f[1])) 
  192.         {
  193.           m_state=STATE_CONNECTED;
  194.         }
  195.         else if (FD_ISSET(m_socket,&f[2]))
  196.         {
  197.           m_errorstr="connecting to host";
  198.           m_state=STATE_ERROR;
  199.         }
  200.       }
  201.     break;
  202.     case STATE_CONNECTED:
  203.     case STATE_CLOSING:
  204.       if (m_send_len>0 && bytes_allowed_to_send>0)
  205.       {
  206.         int len=m_send_buffer_len-m_send_pos;
  207.         if (len > m_send_len) len=m_send_len;
  208.         if (len > bytes_allowed_to_send) len=bytes_allowed_to_send;
  209.         if (len > 0)
  210.         {
  211.           int res=::send(m_socket,m_send_buffer+m_send_pos,len,0);
  212.           if (res==-1 && ERRNO != EWOULDBLOCK)
  213.           {            
  214. //            m_state=STATE_CLOSED;
  215. //            return;
  216.           }
  217.           if (res>0)
  218.           {
  219.             bytes_allowed_to_send-=res;
  220.             if (bytes_sent) *bytes_sent+=res;
  221.             m_send_pos+=res;
  222.             m_send_len-=res;
  223.           }
  224.         }
  225.         if (m_send_pos>=m_send_buffer_len) 
  226.         {
  227.           m_send_pos=0;
  228.           if (m_send_len>0)
  229.           {
  230.             len=m_send_buffer_len-m_send_pos;
  231.             if (len > m_send_len) len=m_send_len;
  232.             if (len > bytes_allowed_to_send) len=bytes_allowed_to_send;
  233.             int res=::send(m_socket,m_send_buffer+m_send_pos,len,0);
  234.             if (res==-1 && ERRNO != EWOULDBLOCK)
  235.             {
  236. //              m_state=STATE_CLOSED;
  237.             }
  238.             if (res>0)
  239.             {
  240.               bytes_allowed_to_send-=res;
  241.               if (bytes_sent) *bytes_sent+=res;
  242.               m_send_pos+=res;
  243.               m_send_len-=res;
  244.             }
  245.           }
  246.         }
  247.       }
  248.       if (m_recv_len<m_recv_buffer_len)
  249.       {
  250.         int len=m_recv_buffer_len-m_recv_pos;
  251.         if (len > m_recv_buffer_len-m_recv_len) len=m_recv_buffer_len-m_recv_len;
  252.         if (len > bytes_allowed_to_recv) len=bytes_allowed_to_recv;
  253.         if (len>0)
  254.         {
  255.           int res=::recv(m_socket,m_recv_buffer+m_recv_pos,len,0);
  256.           if (res == 0 || (res < 0 && ERRNO != EWOULDBLOCK))
  257.           {        
  258.             m_state=STATE_CLOSED;
  259.             break;
  260.           }
  261.           if (res > 0)
  262.           {
  263.             bytes_allowed_to_recv-=res;
  264.             if (bytes_rcvd) *bytes_rcvd+=res;
  265.             m_recv_pos+=res;
  266.             m_recv_len+=res;
  267.           }
  268.         }
  269.         if (m_recv_pos >= m_recv_buffer_len)
  270.         {
  271.           m_recv_pos=0;
  272.           if (m_recv_len < m_recv_buffer_len)
  273.           {
  274.             len=m_recv_buffer_len-m_recv_len;
  275.             if (len > bytes_allowed_to_recv) len=bytes_allowed_to_recv;
  276.             if (len > 0)
  277.             {
  278.               int res=::recv(m_socket,m_recv_buffer+m_recv_pos,len,0);
  279.               if (res == 0 || (res < 0 && ERRNO != EWOULDBLOCK))
  280.               {        
  281.                 m_state=STATE_CLOSED;
  282.                 break;
  283.               }
  284.               if (res > 0)
  285.               {
  286.                 bytes_allowed_to_recv-=res;
  287.                 if (bytes_rcvd) *bytes_rcvd+=res;
  288.                 m_recv_pos+=res;
  289.                 m_recv_len+=res;
  290.               }
  291.             }
  292.           }
  293.         }
  294.       }
  295.       if (m_state == STATE_CLOSING)
  296.       {
  297.         if (m_send_len < 1) m_state = STATE_CLOSED;
  298.       }
  299.     break;
  300.     default: break;
  301.   }
  302. }
  303.  
  304. void JNL_Connection::close(int quick)
  305. {
  306.   if (quick || m_state == STATE_RESOLVING || m_state == STATE_CONNECTING)
  307.   {
  308.     m_state=STATE_CLOSED;
  309.     if (m_socket >= 0)
  310.     {
  311.       ::shutdown(m_socket, SHUT_RDWR);
  312.       ::closesocket(m_socket);
  313.     }
  314.     m_socket=-1;
  315.     memset(m_recv_buffer,0,m_recv_buffer_len);
  316.     memset(m_send_buffer,0,m_send_buffer_len);
  317.     m_remote_port=0;
  318.     m_recv_len=m_recv_pos=0;
  319.     m_send_len=m_send_pos=0;
  320.     m_host[0]=0;
  321.     memset(&m_saddr,0,sizeof(m_saddr));
  322.   }
  323.   else
  324.   {
  325.     if (m_state == STATE_CONNECTED) m_state=STATE_CLOSING;
  326.   }
  327. }
  328.  
  329. int JNL_Connection::send_bytes_in_queue(void)
  330. {
  331.   return m_send_len;
  332. }
  333.  
  334. int JNL_Connection::send_bytes_available(void)
  335. {
  336.   return m_send_buffer_len-m_send_len;
  337. }
  338.  
  339. int JNL_Connection::send(char *data, int length)
  340. {
  341.   if (length > send_bytes_available())
  342.   {
  343.     return -1;
  344.   }
  345.   
  346.   int write_pos=m_send_pos+m_send_len;
  347.   if (write_pos >= m_send_buffer_len) 
  348.   {
  349.     write_pos-=m_send_buffer_len;
  350.   }
  351.  
  352.   int len=m_send_buffer_len-write_pos;
  353.   if (len > length) 
  354.   {
  355.     len=length;
  356.   }
  357.  
  358.   memcpy(m_send_buffer+write_pos,data,len);
  359.   if (length > len)
  360.   {
  361.     memcpy(m_send_buffer,data+len,length-len);
  362.   }
  363.   m_send_len+=length;
  364.   return 0;
  365. }
  366.  
  367. int JNL_Connection::send_string(char *line)
  368. {
  369.   return send(line,strlen(line));
  370. }
  371.  
  372. int JNL_Connection::recv_bytes_available(void)
  373. {
  374.   return m_recv_len;
  375. }
  376.  
  377. int JNL_Connection::peek_bytes(char *data, int maxlength)
  378. {
  379.   if (maxlength > m_recv_len)
  380.   {
  381.     maxlength=m_recv_len;
  382.   }
  383.   int read_pos=m_recv_pos-m_recv_len;
  384.   if (read_pos < 0) 
  385.   {
  386.     read_pos += m_recv_buffer_len;
  387.   }
  388.   int len=m_recv_buffer_len-read_pos;
  389.   if (len > maxlength)
  390.   {
  391.     len=maxlength;
  392.   }
  393.   memcpy(data,m_recv_buffer+read_pos,len);
  394.   if (len < maxlength)
  395.   {
  396.     memcpy(data+len,m_recv_buffer,maxlength-len);
  397.   }
  398.  
  399.   return maxlength;
  400. }
  401.  
  402. int JNL_Connection::recv_bytes(char *data, int maxlength)
  403. {
  404.   
  405.   int ml=peek_bytes(data,maxlength);
  406.   m_recv_len-=ml;
  407.   return ml;
  408. }
  409.  
  410. int JNL_Connection::getbfromrecv(int pos, int remove)
  411. {
  412.   int read_pos=m_recv_pos-m_recv_len + pos;
  413.   if (pos < 0 || pos > m_recv_len) return -1;
  414.   if (read_pos < 0) 
  415.   {
  416.     read_pos += m_recv_buffer_len;
  417.   }
  418.   if (read_pos >= m_recv_buffer_len)
  419.   {
  420.     read_pos-=m_recv_buffer_len;
  421.   }
  422.   if (remove) m_recv_len--;
  423.   return m_recv_buffer[read_pos];
  424. }
  425.  
  426. int JNL_Connection::recv_lines_available(void)
  427. {
  428.   int l=recv_bytes_available();
  429.   int lcount=0;
  430.   int lastch=0;
  431.   int pos;
  432.   for (pos=0; pos < l; pos ++)
  433.   {
  434.     int t=getbfromrecv(pos,0);
  435.     if (t == -1) return lcount;
  436.     if ((t=='\r' || t=='\n') &&(
  437.          (lastch != '\r' && lastch != '\n') || lastch==t
  438.         )) lcount++;
  439.     lastch=t;
  440.   }
  441.   return lcount;
  442. }
  443.  
  444. int JNL_Connection::recv_line(char *line, int maxlength)
  445. {
  446.   if (maxlength > m_recv_len) maxlength=m_recv_len;
  447.   while (maxlength--)
  448.   {
  449.     int t=getbfromrecv(0,1);
  450.     if (t == -1) 
  451.     {
  452.       *line=0;
  453.       return 0;
  454.     }
  455.     if (t == '\r' || t == '\n')
  456.     {
  457.       int r=getbfromrecv(0,0);
  458.       if ((r == '\r' || r == '\n') && r != t) getbfromrecv(0,1);
  459.       *line=0;
  460.       return 0;
  461.     }
  462.     *line++=(char)t;
  463.   }
  464.   return 1;
  465. }
  466.  
  467. unsigned long JNL_Connection::get_interface(void)
  468. {
  469.   if (m_socket==-1) return 0;
  470.   struct sockaddr_in sin;
  471.   memset(&sin,0,sizeof(sin));
  472.   socklen_t len=16;
  473.   if (::getsockname(m_socket,(struct sockaddr *)&sin,&len)) return 0;
  474.   return (unsigned long) sin.sin_addr.s_addr;
  475. }
  476.